﻿using System.Collections;
using System.Reflection;
using Furion;
using HiSportApplet.Core;
using HiSportApplet.Core.Cache;
using HiSportApplet.Core.Const;
using HiSportApplet.Core.Entity;
using HiSportApplet.Core.Options;
using HiSportApplet.SqlSugar;
using Mapster;
using Microsoft.Extensions.DependencyInjection;
using SqlSugar;
using Yitter.IdGenerator;

namespace HiSportApplet.Application.Common.SqlSugar
{
    public static class SqlSugarSetup
    {
        /// <summary>
        /// SqlSugar 上下文初始化
        /// </summary>
        /// <param name="services"></param>
        public static void AddSqlSugar(this IServiceCollection services)
        {
            //获取所有数据库选项配置
            var dbOptions = App.GetOptions<DbConnectionOptions>();
            //配置链接属性
            dbOptions.ConnectionConfigs.ForEach(SetDbConfig);

            SqlSugarScope sqlSugar = new(dbOptions.ConnectionConfigs.Adapt<List<ConnectionConfig>>(), db =>
            {
                dbOptions.ConnectionConfigs.ForEach(config =>
                {
                    var dbProvider = db.GetConnectionScope(config.ConfigId);
                    SetDbAop(dbProvider, dbOptions.EnableConsoleSql);

                });
            });

            services.AddSingleton<ISqlSugarClient>(sqlSugar); // 单例注册
            services.AddScoped(typeof(SqlSugarRepository<>)); // 仓储注册
            services.AddUnitOfWork<SqlSugarUnitOfWork>(); // 事务与工作单元注册
        }

        /// <summary>
        /// 配置连接属性
        /// </summary>
        /// <param name="config"></param>
        public static void SetDbConfig(DbConnectionConfig config)
        {
            var configureExternalServices = new ConfigureExternalServices
            {
                EntityNameService = (type, entity) => // 处理表
                {
                    entity.IsDisabledDelete = true; // 禁止删除非 sqlsugar 创建的列
                                                    // 只处理贴了特性[SugarTable]表
                    if (!type.GetCustomAttributes<SugarTable>().Any())
                        return;
                    if (config.DbSettings.EnableUnderLine && !entity.DbTableName.Contains('_'))
                        entity.DbTableName = UtilMethods.ToUnderLine(entity.DbTableName); // 驼峰转下划线
                },
                EntityService = (type, column) => // 处理列
                {
                    // 只处理贴了特性[SugarColumn]列
                    if (!type.GetCustomAttributes<SugarColumn>().Any())
                        return;
                    if (new NullabilityInfoContext().Create(type).WriteState is NullabilityState.Nullable)
                        column.IsNullable = true;
                    if (config.DbSettings.EnableUnderLine && !column.IsIgnore && !column.DbColumnName.Contains('_'))
                        column.DbColumnName = UtilMethods.ToUnderLine(column.DbColumnName); // 驼峰转下划线

                },
                DataInfoCacheService = new SqlSugarCache(),
            };
            config.ConfigureExternalServices = configureExternalServices;
            config.InitKeyType = InitKeyType.Attribute;
            config.IsAutoCloseConnection = true;
            config.MoreSettings = new ConnMoreSettings
            {
                IsAutoRemoveDataCache = true,
                IsAutoDeleteQueryFilter = true, // 启用删除查询过滤器
                IsAutoUpdateQueryFilter = true, // 启用更新查询过滤器
                SqlServerCodeFirstNvarchar = true // 采用Nvarchar
            };
        }

        /// <summary>
        /// 配置Aop
        /// </summary>
        /// <param name="db"></param>
        /// <param name="enableConsoleSql"></param>
        public static void SetDbAop(SqlSugarScopeProvider db, bool enableConsoleSql)
        {
            var config = db.CurrentConnectionConfig;

            // 设置超时时间
            db.Ado.CommandTimeOut = 30;

            // 打印SQL语句
            if (enableConsoleSql)
            {
                db.Aop.OnLogExecuting = (sql, pars) =>
                {
                    var originColor = Console.ForegroundColor;
                    if (sql.StartsWith("SELECT", StringComparison.OrdinalIgnoreCase))
                        Console.ForegroundColor = ConsoleColor.Green;
                    if (sql.StartsWith("UPDATE", StringComparison.OrdinalIgnoreCase) || sql.StartsWith("INSERT", StringComparison.OrdinalIgnoreCase))
                        Console.ForegroundColor = ConsoleColor.Yellow;
                    if (sql.StartsWith("DELETE", StringComparison.OrdinalIgnoreCase))
                        Console.ForegroundColor = ConsoleColor.Red;
                    Console.WriteLine("【" + DateTime.Now + "——执行SQL】\r\n" + UtilMethods.GetSqlString(config.DbType, sql, pars) + "\r\n");
                    Console.ForegroundColor = originColor;
                    App.PrintToMiniProfiler("SqlSugar", "Info", sql + "\r\n" + db.Utilities.SerializeObject(pars.ToDictionary(it => it.ParameterName, it => it.Value)));
                };
                db.Aop.OnError = ex =>
                {
                    if (ex.Parametres == null) return;
                    var originColor = Console.ForegroundColor;
                    Console.ForegroundColor = ConsoleColor.DarkRed;
                    var pars = db.Utilities.SerializeObject(((SugarParameter[])ex.Parametres).ToDictionary(it => it.ParameterName, it => it.Value));
                    Console.WriteLine("【" + DateTime.Now + "——错误SQL】\r\n" + UtilMethods.GetSqlString(config.DbType, ex.Sql, (SugarParameter[])ex.Parametres) + "\r\n");
                    Console.ForegroundColor = originColor;
                    App.PrintToMiniProfiler("SqlSugar", "Error", $"{ex.Message}{Environment.NewLine}{ex.Sql}{pars}{Environment.NewLine}");
                };
                db.Aop.OnLogExecuted = (sql, pars) =>
                {
                    // 执行时间超过5秒
                    if (db.Ado.SqlExecutionTime.TotalSeconds > 5)
                    {
                        var fileName = db.Ado.SqlStackTrace.FirstFileName; // 文件名
                        var fileLine = db.Ado.SqlStackTrace.FirstLine; // 行号
                        var firstMethodName = db.Ado.SqlStackTrace.FirstMethodName; // 方法名
                        var log = $"【所在文件名】：{fileName}\r\n【代码行数】：{fileLine}\r\n【方法名】：{firstMethodName}\r\n" + $"【sql语句】：{UtilMethods.GetSqlString(config.DbType, sql, pars)}";
                        var originColor = Console.ForegroundColor;
                        Console.ForegroundColor = ConsoleColor.DarkYellow;
                        Console.WriteLine(log);
                        Console.ForegroundColor = originColor;
                    }
                };
            }
            // 数据审计
            db.Aop.DataExecuting = (oldValue, entityInfo) =>
            {

                if (entityInfo.OperationType == DataFilterType.InsertByObject)
                {
                    // 主键(long类型)且没有值的---赋值雪花Id
                    if (entityInfo.EntityColumnInfo.IsPrimarykey && entityInfo.EntityColumnInfo.PropertyInfo.PropertyType == typeof(long))
                    {
                        var id = entityInfo.EntityColumnInfo.PropertyInfo.GetValue(entityInfo.EntityValue);
                        if (id == null || (long)id == 0)
                            entityInfo.SetValue(YitIdHelper.NextId());
                    }

                    if (entityInfo.PropertyName == "CreateTime")
                        entityInfo.SetValue(DateTime.Now);
                    if (entityInfo.PropertyName == "IsDelete")
                        entityInfo.SetValue(false);
                    if (entityInfo.PropertyName == "UpdateTime")
                        entityInfo.SetValue("3000-01-01");
                    if (entityInfo.PropertyName == "UpdateUserId")
                        entityInfo.SetValue("0");
                    if (App.User != null)
                    {
                        if (entityInfo.PropertyName == "TenantId")
                        {
                            var tenantId = ((dynamic)entityInfo.EntityValue).TenantId;
                            if (tenantId == null || tenantId == 0)
                                entityInfo.SetValue(App.User.FindFirst(ClaimConst.TenantId)?.Value);
                        }
                        if (entityInfo.PropertyName == "CreateUserId")
                        {
                            var createUserId = ((dynamic)entityInfo.EntityValue).CreateUserId;
                            if (createUserId == 0 || createUserId == null)
                                entityInfo.SetValue(App.User.FindFirst(ClaimConst.UserId)?.Value);
                        }

                        if (entityInfo.PropertyName == "OrgId")
                        {
                            var OrgId = ((dynamic)entityInfo.EntityValue).OrgId;
                            if (OrgId == 0 || OrgId == null)
                                entityInfo.SetValue(App.User.FindFirst(ClaimConst.OrgId)?.Value);
                        }
                    }
                }
                if (entityInfo.OperationType == DataFilterType.UpdateByObject)
                {
                    if (entityInfo.PropertyName == "UpdateTime")
                        entityInfo.SetValue(DateTime.Now);
                    if (entityInfo.PropertyName == "UpdateUserId")
                        entityInfo.SetValue(App.User?.FindFirst(ClaimConst.UserId)?.Value);
                }
            };

            

            // 配置假删除过滤器
            db.QueryFilter.AddTableFilter<IDeletedFilter>(u => u.IsDelete == false);


            // 配置自定义过滤器
            SqlSugarFilter.SetCustomEntityFilter(db);
        }


    }
}
